/**
 * 版权所有 2009-2012山东新北洋信息技术股份有限公司
 * 保留所有权利。
 */
package com.linyaonan.leetcode.easy._104;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;

/**
 *
 * 给定一个二叉树，找出其最大深度。
 *
 * 二叉树的深度为根节点到最远叶子节点的最长路径上的节点数。
 *
 * 说明: 叶子节点是指没有子节点的节点。
 *
 * 示例：
 * 给定二叉树 [3,9,20,null,null,15,7]，
 *
 *     3
 *    / \
 *   9  20
 *     /  \
 *    15   7
 * 返回它的最大深度 3 。
 *
 *
 * @ProjectName: leetcode
 * @Package: com.linyaonan.leetcode.easy._104
 * @ClassName: MaxDepthTree
 * @Author: linyaonan
 * @Date: 2019/10/25 14:50
 */
public class MaxDepthTree {

    class MaxNum {
        int max;
        private void setMax(int num) {
            max = Math.max(num, max);
        }
    }

    public int maxDepth(TreeNode root) {
        if (root == null) {
            return 0;
        }
        MaxNum maxNum = new MaxNum();
        getMaxDepth(root, maxNum, 0);
        return maxNum.max;
    }

    public int maxDepth2(TreeNode root) {
        if (root == null) {
            return 0;
        }
        // 层序遍历统计层数
        LinkedList<TreeNode> stack = new LinkedList<>();

        stack.addLast(root);

        int ceng = 0;

        while (!stack.isEmpty()) {
            LinkedList<TreeNode> next = new LinkedList<>();
            for (TreeNode treeNode : stack) {
                if (treeNode != null) {
                    if (treeNode.left != null) {
                        next.addLast(treeNode.left);
                    }
                    if (treeNode.right != null) {
                        next.addLast(treeNode.right);
                    }
                }
            }
            stack = next;
            ceng++;
        }

        return ceng;

    }

    private void getMaxDepth(TreeNode node, MaxNum maxNum, int index) {
        index++;
        if (node.left != null) {
            getMaxDepth(node.left, maxNum, index);
        } else {
            maxNum.setMax(index);
        }
        if (node.right != null) {
            getMaxDepth(node.right, maxNum, index);
        } else {
            maxNum.setMax(index);
        }
    }

    public int maxDepth3(TreeNode root) {
        return getMax(root, 0);
    }

    private int getMax(TreeNode root, int l) {
        if (root == null) {
            return l;
        }
        int le = getMax(root.left, l+1);
        int ri = getMax(root.right, l+1);
        return Math.max(le, ri);
    }

}
